Skip to content

Add Diffusion Download, Inference and Trainer #306

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Open
wants to merge 67 commits into
base: main
Choose a base branch
from
Open

Add Diffusion Download, Inference and Trainer #306

wants to merge 67 commits into from

Conversation

deep1401
Copy link
Contributor

@deep1401 deep1401 commented May 21, 2025 •
edited
Loading

  • In order to preview image datasets you would need the add/multimodal-training (API) and add/preview-image-data (App) branches merged or switched on
  • Trial dataset for trainer: datasets-examples/doc-image-6
  • Added download and inference abilities too

@codecov Codecov
Copy link

codecov bot commented May 21, 2025 •
edited
Loading

Codecov Report

Attention: Patch coverage is 44.37401% with 351 lines in your changes missing coverage. Please review.

Files with missing lines Patch % Lines
transformerlab/routers/diffusion.py 51.97% 216 Missing and 39 partials ⚠️
transformerlab/models/huggingfacemodel.py 0.00% 47 Missing ⚠️
transformerlab/routers/jobs.py 9.75% 37 Missing ⚠️
transformerlab/routers/model.py 0.00% 12 Missing ⚠️

📢 Thoughts on this report? Let us know!

@deep1401 deep1401 changed the title Add a Stable Diffusion Trainer Add Diffusion Download, Inference and Trainer May 21, 2025
@deep1401 deep1401 mentioned this pull request May 21, 2025
Open
github-advanced-security[bot]
github-advanced-security bot found potential problems May 28, 2025
View reviewed changes
deep1401 and others added 17 commits May 29, 2025 10:49
@deep1401
Make checking stable diffusion logic better
76cde89
@deep1401
Simply is_stable_diffusion check
eaeb093
@deep1401
Add generation_time to image history
df3a9ad
@deep1401
Add support for all prominent architectures in diffusion
3ee008c
@deep1401
Change the name of the allowed architectures list
19a8995
@deep1401
Add ability for img2img generation
3287b06
@deep1401
Add img2img architectures in the allowed list
ebb9a69
@deep1401
Merge branch 'main' into add/diffusion-trainer
7c93605
@deep1401
Ruff fixes
ed14aef
@deep1401
Fix broken tests and add new ones for is_stable_diffusion
1c06784
@deep1401
Fix dataset creation for img2img
acfee55
@deep1401
Remove cachine mechanism and add pipeline cleanup
49f2fd8
@deep1401
Modify trainer and modify lora loader for sdxl
995c4e4
@deep1401
Ruff fixes
3d335ce
@deep1401
Stricter checks for img2img and blank model_id
2151ea3
@deep1401
Merge branch 'add/diffusion-trainer' of https://github.com/transforme…
0f3ed72 
…rlab/transformerlab-api into add/diffusion-trainer
@deep1401
Generate multiple images (gen+history)
d770793
github-advanced-security[bot]
github-advanced-security bot found potential problems May 30, 2025
View reviewed changes
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Reviewers

@aliasaria aliasaria Awaiting requested review from aliasaria

@dadmobile dadmobile Awaiting requested review from dadmobile

@mina-parham mina-parham Awaiting requested review from mina-parham

@aahaanmaini aahaanmaini Awaiting requested review from aahaanmaini

@Sourenm Sourenm Awaiting requested review from Sourenm

At least 1 approving review is required to merge this pull request.

Assignees
No one assigned
Labels
None yet
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
2 participants
@deep1401 @aliasaria